import codecs
import os


def sensitive_word(sensitive_word_dir):
    """
    对word_list进行敏感词判断
    :return: exist  用于标记文章里是否有敏感词汇，默认1为有，2为无
    """
    sensitive_vocabulary = []  # 所有敏感词汇组合的list
    # 跟进敏感词目录依次打开目录中文件，并添加到敏感词汇list里
    if os.path.isdir(sensitive_word_dir):  # 判断sensitive_word_dir是否为目录
        file_names = os.listdir(sensitive_word_dir)
        for filename in file_names:
            file = codecs.open(sensitive_word_dir + '\\' + filename)
            for line in file:
                line = line.strip()
                sensitive_vocabulary.append(line)
    return sensitive_vocabulary
